/**
 * 植物病害信息服务
 * 负责从数据库获取植物病害信息
 */

/**
 * 根据植物名称获取病害信息
 * @param {object} pool - MySQL连接池
 * @param {string} plantName - 植物名称
 * @returns {Promise<object>} - 返回病害信息
 */
async function getDiseaseInfo(pool, plantName) {
  try {
    // 精确匹配
    let [rows] = await pool.query(
      'SELECT * FROM plant_diseases WHERE plant_name = ? LIMIT 1',
      [plantName]
    );
    
    // 如果没有精确匹配，尝试模糊匹配
    if (rows.length === 0) {
      [rows] = await pool.query(
        'SELECT * FROM plant_diseases WHERE plant_name LIKE ? LIMIT 1',
        [`%${plantName}%`]
      );
    }
    
    if (rows.length > 0) {
      return {
        success: true,
        treatment: rows[0].treatment
      };
    } else {
      // 返回默认信息
      return {
        success: true,
        treatment: `暂无${plantName}的病害防治信息，建议咨询专业农技人员。`
      };
    }
  } catch (err) {
    console.error('查询数据库错误:', err);
    throw err;
  }
}

module.exports = {
  getDiseaseInfo
};
